package controllers

import (
	"github.com/astaxie/beego"
	"github.com/beego/i18n"
)

// I18nController 多语言控制器
type I18nController struct {
	beego.Controller
	i18n.Locale
}

// Prepare 初始化之后请求之前的准备逻辑
func (i18nCtl *I18nController) Prepare() {
	lang := i18nCtl.GetString("lang")
	if lang == "zh-CN" {
		i18nCtl.Lang = lang
	} else {
		i18nCtl.Lang = "en-US"
	}
	i18nCtl.Data["Lang"] = i18nCtl.Lang
}

// Get 多语言控制器的get请求处理
func (i18nCtl *I18nController) Get() {
	i18nCtl.Data["Website"] = "beego.me"
	i18nCtl.Data["Email"] = "astaxie@gmail.com"
	i18nCtl.Data["Hi"] = i18nCtl.Tr("hi")
	i18nCtl.Data["Bye"] = i18nCtl.Tr("bye")
	i18nCtl.Data["About"] = "about"
	i18nCtl.TplName = "i18n.html"
}
